How can I create a type with keys from union type and different value types